how to initialize an array python